from datetime import date
from openai import OpenAI

ds = OpenAI(api_key="sk-HBKZOTbyk9no5N1dbjfKw60kR9pXNKHImbTJHtdGfuf7gwB2", base_url="https://api.lkeap.cloud.tencent.com/v1")

req_msg = [
    {"role": "system", "content": "你是一个期货行情分析员，我为您提供行情数据你需要根据数据给出买卖信号，注意您只输出-1(卖出)0(观望)+1(买入)，不要输出其他描述信息"},
    {"role": "system", "content": "最近20日的K线数据DK，最近30小时的K线数据HK，以及当前持仓POSITION"},
    {"role": "system", "content": "30分钟级别：DMA，ATR，BOLL,RSI，KDJ，MACD"}, 
    {"role": "system", "content": "15分钟级别：DMA，ATR，BOLL,RSI，KDJ，MACD"}, 
    {"role": "system", "content": "5分钟级别：ASI，MFI，DFMA，BBI，ATR，CCI，PSY，BOLL，BIAS，RSI，KDJ，MACD"},
    {"role": "system", "content": "1分钟级别：ASI，MFI，DFMA，BBI，ATR，CCI，PSY，BOLL，BIAS，RSI，KDJ，MACD"},
    {"role": "system", "content": "注意你只能通过上面提供的数据不能基于假设做出判断，控制每日交易次数避免频繁交易，控制日内回撤，过滤掉震荡周期"},
    {'role': 'user', 'content': "DK (last 20 days)：\n    open  high   low  close  volume\n9   7817  7830  7761   7824  298461\n10  7832  7866  7698   7788  591524\n11  7670  7696  7589   7608  619156\n12  7605  7691  7600   7689  582146\n13  7699  7884  7683   7872  866528\n14  7853  7924  7804   7833  722584\n15  7820  7824  7702   7770  789980\n16  7728  7822  7688   7783  820021\n17  7770  7865  7746   7812  733675\n18  7803  7819  7717   7720  702753\n19  7667  7794  7607   7778  868528\n20  7786  7826  7640   7687  736463\n21  8069  8125  8001   8107  432032\n22  8120  8121  8035   8042  497965\n23  8040  8091  7977   8055  526165\n24  8055  8146  8006   8062  903108\n25  8085  8105  7925   7972  801858\n26  7916  7995  7839   7967  700031\n27  7940  8055  7899   8028  678940\n28  8028  8263  7975   8254  894058\n\nHK (last 30 hours)：\n    open  high   low  close  volume\n29  7957  7967  7949   7952   31537\n30  7952  7972  7945   7966   21795\n31  7966  7970  7955   7969   10530\n32  7944  7951  7921   7944   66602\n33  7944  7995  7944   7965   51290\n34  7965  7970  7958   7963   15496\n35  7942  7958  7939   7950   18970\n36  7950  7977  7949   7967   34399\n37  7940  7987  7899   7986  200327\n38  7986  8010  7965   7993  120933\n39  7993  8018  7978   8013   75814\n40  8013  8029  7997   8021   40760\n41  8021  8043  8017   8037   35998\n42  8037  8038  8021   8025   13500\n43  8035  8042  7997   8013   74083\n44  8013  8028  8002   8021   27181\n45  8021  8035  8010   8027   18487\n46  8036  8055  8027   8030   32271\n47  8030  8032  8015   8028   39586\n48  8028  8038  7985   7988  139234\n49  7988  8006  7975   7989   80714\n50  7989  8009  7981   7999   68414\n51  7999  8006  7982   7984   34167\n52  7984  7999  7983   7997   18498\n53  7997  8007  7994   7999   11498\n54  8023  8119  8023   8094  168890\n55  8094  8096  8076   8093   32044\n56  8093  8099  8086   8092   15300\n57  8098  8170  8096   8169   99114\n58  8169  8263  8165   8254  226185\n\nPOSITION：\n0.3\n\n30分钟周期数据(last 20)：\n     open  high   low  close  volume  ddd  ama   tr  atr   mid   top  bottom  rsi   k   d    j  diff  dea  bar\n99   8030  8030  8015   8028   23265    5    8   15   22  7997  8059    7934   61  56  57   52    12   10    4\n100  8028  8038  7994   8016   64976    6    8   44   23  8000  8059    7940   49  49  55   38    11   10    1\n101  8016  8032  7985   7988   74258    2    8   47   25  8001  8058    7944   31  34  48    7     8   10   -4\n102  7988  8006  7976   7983   44120   -5    8   30   26  8001  8057    7946   29  26  40   -4     5    9   -8\n103  7983  8002  7975   7989   36594  -11    7   27   26  8002  8056    7949   35  23  35    0     3    8  -10\n104  7989  8002  7981   7988   37740  -17    6   21   26  8004  8053    7954   35  21  30    2     1    6  -10\n105  7988  8009  7986   7999   30674  -18    4   23   27  8005  8052    7959   46  24  28   16     1    5   -9\n106  7999  8006  7991   7998   19760  -14    2   15   25  8007  8050    7963   46  28  28   28     0    4   -8\n107  7998  7999  7982   7984   14407   -9    1   17   24  8008  8043    7973   35  23  26   17    -1    3   -9\n108  7984  7992  7983   7989    9108   -5    0    9   24  8008  8043    7974   41  23  25   18    -2    2   -8\n109  7989  7999  7983   7997    9390    0   -1   16   24  8008  8043    7974   49  28  26   32    -2    1   -6\n110  7997  8007  7994   7999   11498    2   -2   13   23  8009  8043    7974   51  42  32   64    -2    1   -5\n111  8023  8119  8023   8099  134591   11   -2  120   29  8012  8062    7963   85  57  40   91     7    2    9\n112  8099  8107  8079   8094   34299   22   -2   28   30  8016  8074    7957   82  65  48   99    13    4   17\n113  8094  8096  8076   8084   13531   32    0   20   31  8019  8083    7955   75  68  55   95    16    7   20\n114  8085  8095  8078   8093   18513   40    1   17   29  8022  8092    7951   77  73  61   96    20    9   21\n115  8093  8099  8086   8092   15300   50    4   13   26  8024  8099    7948   76  75  66   94    22   12   21\n116  8098  8170  8096   8169   99114   46    7   78   30  8029  8123    7934   88  83  72  107    30   15   29\n117  8169  8259  8165   8232  154242   49    9   94   35  8037  8160    7913   92  86  76  104    41   20   40\n118  8232  8263  8229   8254   71943   57   12   34   36  8046  8196    7897   93  89  81  107    50   26   48\n\n15分钟周期数据(last 20)：\n     open  high   low  close  volume  ddd  ama   tr  atr   mid   top  bottom  rsi   k   d    j  diff  dea  bar\n99   7990  7992  7985   7989    3274   -2   -4    7   16  8006  8040    7972   42  34  40   22    -6   -4   -2\n100  7989  7992  7983   7988    3842   -3   -4    9   15  8005  8040    7970   41  30  37   17    -6   -5   -2\n101  7988  7999  7988   7997    5548   -4   -5   11   14  8004  8037    7970   53  41  38   46    -5   -5    0\n102  7997  8003  7994   7998    4310   -1   -5    9   13  8002  8032    7972   55  50  42   65    -4   -5    1\n103  7998  8007  7995   7999    7188    0   -4   12   12  8001  8029    7973   56  56  47   74    -4   -4    2\n104  8023  8114  8023   8098   97881   12   -3  115   19  8004  8050    7957   90  66  53   93     5   -3   15\n105  8098  8119  8092   8099   36710   24   -2   27   20  8006  8065    7947   90  73  60   99    12    0   23\n106  8099  8107  8079   8092   23423   34    1   28   20  8009  8077    7941   84  75  65   96    16    3   25\n107  8092  8098  8085   8094   10876   42    3   13   20  8011  8087    7936   84  77  69   94    20    7   26\n108  8094  8096  8076   8084   13531   50    6   20   21  8014  8095    7934   75  76  71   86    21   10   23\n109  8085  8089  8078   8082    8889   37    9   11   21  8017  8102    7932   73  75  73   79    22   12   20\n110  8082  8095  8078   8093    9624   25   10   17   22  8020  8110    7930   77  76  74   81    24   14   18\n111  8093  8099  8090   8096    7430   16   12    9   21  8024  8117    7931   78  78  75   84    25   16   16\n112  8096  8098  8086   8092    7870    6   12   12   21  8028  8124    7931   73  76  75   77    25   18   13\n113  8098  8113  8096   8111   25202    1   12   21   22  8033  8133    7932   80  78  76   81    26   20   13\n114  8111  8170  8111   8169   73912   11   13   59   26  8039  8151    7927   90  85  79   96    32   22   19\n115  8169  8259  8165   8254  106263   28   14   94   32  8049  8188    7911   94  89  82  102    42   26   32\n116  8254  8258  8229   8232   47979   41   16   29   33  8058  8212    7904   83  88  84   95    49   31   36\n117  8232  8259  8229   8252   29691   57   19   30   35  8068  8237    7900   85  91  86   99    54   35   38\n118  8252  8263  8234   8254   42252   69   22   29   28  8078  8259    7897   85  92  88  100    58   40   37\n\n5分钟周期数据(last 24)：\n     open  high   low  close  volume   asi  mfi   bbi  tr  atr  cci  psy  psyma   mid   top  bottom  bias  rsi   k   d    j  diff  dea  bar\n95   8095  8099  8094   8096    3018   153   56  8091   5   10  147   58     43  8081  8144    8018     0   65  78  64  104    15   17   -6\n96   8096  8098  8093   8096    1837   164   50  8093   5    9  115   50     46  8085  8138    8032     0   65  80  70  101    14   17   -5\n97   8096  8098  8094   8095    1671   155   54  8094   4    9   97   50     49  8088  8128    8048     0   62  80  73   95    13   16   -5\n98   8095  8098  8087   8089    3165    23   46  8093  11    9   29   42     49  8092  8109    8075     0   47  71  73   68    12   15   -6\n99   8089  8094  8086   8092    3034     9   47  8092   8    9   14   50     50  8093  8105    8081     0   54  70  72   66    11   14   -6\n100  8098  8111  8096   8102   13696   186   58  8094  19    9  180   58     51  8093  8106    8081     0   69  68  70   64    11   14   -5\n101  8102  8110  8099   8104    5056   246   71  8095  11    9  158   58     51  8093  8107    8080     0   71  70  70   69    11   13   -4\n102  8104  8113  8104   8111    6450   370   80  8098   9    9  178   67     54  8094  8108    8079     0   78  77  73   87    12   13   -2\n103  8111  8146  8111   8139   28813   997   87  8105  35   11  272   67     57  8095  8117    8073     0   89  81  75   92    14   13    2\n104  8139  8165  8135   8155   27831  1352   92  8114  30   12  258   67     61  8097  8130    8065     0   92  83  78   93    17   14    7\n105  8155  8170  8151   8169   17268  1580   93  8124  19   13  207   75     65  8100  8143    8057     0   94  88  81  102    21   15   11\n106  8169  8217  8165   8217   41410  2530   95  8139  52   16  213   75     68  8104  8167    8042     1   96  92  85  107    27   18   18\n107  8217  8237  8206   8235   34211  2816   96  8156  31   18  191   75     71  8110  8191    8029     1   97  94  88  107    33   21   24\n108  8235  8259  8224   8254   30642  3126   96  8174  35   20  167   83     74  8116  8214    8018     1   98  95  91  105    39   24   29\n109  8254  8258  8235   8238   20685  3087   88  8184  23   21  129   83     76  8122  8230    8014     0   84  92  91   95    42   28   27\n110  8238  8245  8231   8243   13952  3019   83  8193  14   22  103   92     81  8128  8245    8010     0   84  92  91   92    44   31   25\n111  8243  8245  8229   8232   13342  2850   79  8199  16   22   82   83     82  8133  8256    8010     0   75  88  90   84    44   34   21\n112  8232  8248  8229   8247   12270  2985   81  8205  19   23   77   83     83  8139  8269    8010     0   79  89  90   87    45   36   18\n113  8247  8248  8240   8244    6223  3037   82  8210   8   23   71   75     83  8145  8279    8012     0   76  88  89   86    45   38   15\n114  8244  8259  8241   8252   11198  3146   82  8217  18   23   72   75     82  8152  8290    8014     0   78  90  89   90    45   39   12\n115  8252  8263  8246   8250   13815  3183   83  8221  17   23   69   67     79  8158  8298    8018     0   76  85  88   80    44   40    8\n116  8250  8250  8234   8241   14505  2943   78  8224  16   24   46   58     74  8164  8305    8024     0   67  71  83   49    43   41    4\n117  8241  8255  8238   8254   13932  3044   77  8229  17   22   59   58     69  8171  8312    8030     0   72  72  79   58    42   41    2\n118  8262  8275  8238   8248   40025  3044   78  8231  37   23   77   50     64  8177  8317    8036     0   66  62  73   39    40   41   -1\n\n1分钟周期数据(last 30)：\n     open  high   low  close  volume   asi  mfi   bbi  tr  atr  cci  psy  psyma   mid   top  bottom  bias  rsi   k   d    j  diff  dea  bar\n89   8247  8248  8243   8247    1429  2994   50  8242   5    8  148   33     35  8240  8252    8227     0   64  78  65  106     9   11   -4\n90   8247  8248  8244   8245    1456  2965   43  8243   4    7  115   25     33  8240  8252    8229     0   59  80  70  101     8   10   -4\n91   8245  8247  8241   8242    1491  2889   38  8242   6    7   60   25     31  8241  8252    8229     0   52  76  72   85     8   10   -4\n92   8242  8246  8240   8243    1258  2872   31  8242   6    7   51   25     29  8241  8252    8230     0   54  75  73   80     7    9   -4\n93   8243  8245  8241   8244     589  2888   35  8242   4    7   51   33     29  8241  8252    8230     0   56  75  74   78     7    9   -4\n94   8244  8244  8241   8243     653  2880   37  8242   3    6   34   33     29  8241  8250    8231     0   53  67  72   58     7    8   -4\n95   8243  8245  8241   8243    1040  2873   42  8242   4    6   35   33     29  8240  8249    8232     0   53  61  68   48     6    8   -4\n96   8243  8253  8243   8252    3421  3053   54  8244  10    6  148   42     32  8241  8250    8231     0   72  72  69   77     6    8   -2\n97   8252  8259  8252   8257    4053  3154   68  8246   7    6  255   42     35  8241  8253    8230     0   78  78  72   89     7    8   -1\n98   8257  8257  8249   8252    2031  3119   69  8248   8    6  148   42     38  8242  8254    8230     0   63  73  72   74     7    7   -1\n99   8252  8257  8246   8246    2333  2974   58  8247  11    6   73   42     39  8242  8254    8230     0   49  59  68   41     6    7   -2\n100  8246  8255  8246   8254    1975  3056   57  8248   9    6   90   42     40  8243  8256    8230     0   62  64  67   59     7    7   -1\n101  8254  8260  8252   8256    3351  3122   66  8249   8    7  132   50     43  8244  8257    8230     0   64  69  67   72     7    7   -1\n102  8256  8263  8251   8254    3453  3113   63  8250  12    7  109   50     44  8244  8258    8230     0   60  66  67   63     7    7   -1\n103  8254  8260  8249   8250    2703  3032   55  8250  11    7   58   50     46  8245  8258    8231     0   51  57  64   45     6    7   -1\n104  8250  8250  8236   8240    5046  2748   48  8248  14    8  -90   42     46  8244  8258    8231     0   36  43  57   16     5    6   -3\n105  8240  8240  8235   8237    2760  2593   46  8246   5    8 -133   33     44  8244  8258    8231     0   32  31  48   -3     4    6   -4\n106  8237  8242  8234   8236    3090  2547   48  8244   8    8 -118   33     43  8244  8258    8230     0   31  23  40  -11     3    5   -5\n107  8236  8241  8235   8238    1323  2569   49  8243   6    8  -99   42     42  8244  8258    8230     0   37  20  33   -7     2    4   -5\n108  8238  8241  8234   8241    2286  2633   54  8243   7    9  -85   42     40  8244  8258    8231     0   44  21  29    6     1    4   -5\n109  8241  8241  8238   8239    1054  2633   54  8242   3    8  -72   33     38  8245  8258    8232     0   40  20  26    8     1    3   -5\n110  8239  8241  8238   8240    1119  2628   50  8242   3    8  -60   42     38  8245  8257    8233     0   43  20  24   12     1    3   -4\n111  8240  8246  8239   8245    3064  2727   49  8243   7    8  -20   50     40  8245  8257    8233     0   55  28  25   32     1    2   -3\n112  8245  8246  8243   8245    3451  2747   58  8243   3    8   -1   42     42  8245  8257    8233     0   55  41  31   63     1    2   -2\n113  8245  8255  8245   8254    5244  2927   69  8245  10    8   69   42     42  8246  8258    8233     0   71  59  40   97     2    2   -1\n114  8262  8270  8256   8261   14864  3072   77  8248  16    8  154   50     43  8246  8260    8233     0   78  65  48   97     3    2    1\n115  8261  8275  8255   8269    7649  3200   79  8252  20    9  163   58     47  8247  8263    8231     0   83  71  56  102     4    3    4\n116  8269  8273  8248   8249    6748  2905   70  8252  25   10   83   58     50  8247  8264    8231     0   49  60  57   65     4    3    2\n117  8249  8250  8238   8241    6779  2563   65  8250  12   10  -24   58     51  8247  8264    8231     0   41  43  52   23     3    3    0\n118  8241  8249  8239   8248    3985  2588   73  8249  10   10   -5   67     56  8247  8264    8231     0   49  37  47   17     3    3    0\n\n"}
]

while True:
    try:
        response = ds.chat.completions.create(
        #model=self.model,
        model="deepseek-r1",
        messages=req_msg,
        stream=False)
        print(response.choices[0].message.content)
    except Exception as e:
        print("Error:", e)
